from functools import wraps
from flask import g,redirect,url_for,session

# 登录装饰器--没登录无法进行的功能前面加上这个直接跳转到登录页面
def login_required(func):
    @wraps(func)
    def inner(*args,**kwargs):
        if g.user:
            return func(*args,**kwargs)
        else:
            return redirect(url_for("login"))
    return inner

def admin_required(func):
    @wraps(func)
    def inner(*args, **kwargs):
        if g.user and g.user.uid == 1:  # 假设用户信息保存在g.user中，uid为用户ID
            return func(*args, **kwargs)
        else:
            return redirect(url_for("products.products"))
    return inner

def login_req(func):
    @wraps(func)
    def inner(*args, **kwargs):
        if session.get('user_id'):
            return redirect("/")
        else:
            return func(*args, **kwargs)
    return inner



